import torch
print("CUDA 可用:", torch.cuda.is_available())
